import easyocr
import torch

# 这个软件对于文字颜色没有固定为黑白  主要是生活照片这些识别比较好

# 判断是否有 GPU 可用
gpu_available = torch.cuda.is_available()
device = 'cuda' if gpu_available else 'cpu'
print(f"使用设备: {device}")

# 创建 Reader，指定 device
reader_ja = easyocr.Reader(['ja'], gpu=gpu_available)

# 测试图片
# img_path = r"C:\Users\33720\Desktop\17-36-26-032.jpg"
img_path = r"C:\Users\33720\Desktop\5.png"

# # 日文识别
# results_ja = reader_ja.readtext(img_path)
# for (bbox, text, prob) in results_ja:
#     # print(f"日文识别: {text}, 置信度: {prob:.2f}")
#     print(f"{text}")



# 中文+英文识别
# reader_ch_en = easyocr.Reader(['ch_sim','en'])
reader_ch_en = easyocr.Reader(['ja','en'])



# 中文+英文识别
results_ch = reader_ch_en.readtext(img_path)
for (bbox, text, prob) in results_ch:
    # print(f"中文/英文识别: {text}, 置信度: {prob:.2f}")
    print(f"{text}")
